from model.poolnet import build_model
import torch
from torch import nn
from torch.nn import init
import torch.nn.functional as F
import numpy as np
if __name__ == '__main__':
    model = build_model('resnet18')
    model.eval()
    input = torch.randn(1, 3, 512, 512)
    output = model(input)
    print(output.size())